from pandas import DataFrame
from multi_factor_strategy import MultiFactorStrategy

# 初始化策略
strategy = MultiFactorStrategy('你的tushare token')

# 选出前20只股票
selected_stocks:DataFrame = strategy.select_stocks(stock_num=20)

if selected_stocks is not None:
    print("\n选股结果：")
    print(selected_stocks.to_string(index=False))
    
    # 输出因子得分的统计信息
    print("\n因子得分统计：")
    print(selected_stocks['final_score'].describe())
